import torch
from torchsummary import summary
from models.MIMOUNet import build_net
model = build_net("MIMO-UNet-SNN-FIRE")
summary(model.cuda(), input_size=(3, 180, 240), batch_size=-1)
